package com.tang.wx.utils;

import org.apache.commons.lang3.StringUtils;
import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class WxMsgUtil {

    private static String template = "<xml><ToUserName>%s</ToUserName>" +
            "<FromUserName>%s</FromUserName>\n" +
            "<CreateTime>1708147478</CreateTime>\n" +
            "<MsgType><![CDATA[text]]></MsgType>\n" +
            "<Content><![CDATA[%s]]></Content>\n" +
            "<MsgId>24454790042132800</MsgId>\n" +
            "</xml>";

    /**
     * 解析微信发来的请求（XML）.
     *
     * @param msg 消息
     * @return map
     */
    public static Map<String, String> parseXml(final String msg) {
        // 将解析结果存储在HashMap中
        Map<String, String> map = new HashMap<String, String>();

        // 从request中取得输入流
        try (InputStream inputStream = new ByteArrayInputStream(msg.getBytes(StandardCharsets.UTF_8))) {
            // 读取输入流
            SAXReader reader = new SAXReader();
            Document document = reader.read(inputStream);
            // 得到xml根元素
            Element root = document.getRootElement();
            // 得到根元素的所有子节点
            List<Element> elementList = root.elements();

            // 遍历所有子节点
            for (Element e : elementList) {
                map.put(e.getName(), e.getText());
            }
        } catch (Exception e) {
            e.printStackTrace();
        }

        return map;
    }

    /**
     * 构建微信消息
     *
     * @param text 回复的内容
     * @return java.lang.String
     * @author Tang
     * @date 2024/2/17 13:29
     **/
    public static String buildTextMsg(String fromUser, String toUser, String text) {
        return String.format(template, fromUser, toUser, text);
    }

    /**
     * 构建微信消息
     *
     * @param text 回复的内容
     * @return java.lang.String
     * @author Tang
     * @date 2024/2/17 13:29
     **/
    public static String buildTextMsg(Map<String, String> maps, String text) {
        return String.format(template, maps.get("FromUserName"), maps.get("ToUserName"), text);
    }

    /**
     * 获取消息的类型
     *
     * @param map 消息解析出来的参数
     * @return java.lang.String
     * @author Tang
     * @date 2024/2/17 14:05
     **/
    public static String getMsgType(Map<String, String> map) {
        String msgType = map.get("MsgType");
        String event = map.getOrDefault("Event", "");
        return StringUtils.isBlank(event) ? msgType : msgType + "." + event;
    }
}
